package com.xu.rag.pojo.result;

import lombok.Data;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.util.List;

/**
 * @Author: xdl
 * @CreateTime: 2025-01-15 22:31
 * @Description:
 */
@Data
@Accessors(chain = true)
public class TongyiResult implements Serializable {
    private String id;
    private String model;
    private String object;
    private Usage usage;
    private List<EmbeddingData> data;

    @Data
    public static class Usage implements Serializable {
        private String prompt_tokens;
        private String total_tokens;
    }

    @Data
    public static class EmbeddingData implements Serializable {
        private double[] embedding;
        private Integer index;
        private String object;

        private String prompt;
        private String requestId;
    }
}
